// 数据库初始化
const cloud = require('wx-server-sdk');

cloud.init({
  env: cloud.DYNAMIC_CURRENT_ENV
});

const db = cloud.database();

// 确保数据库集合存在
async function ensureCollectionsExist() {
  try {
    // 获取所有集合
    const collections = await db.listCollections().get();
    const collectionNames = collections.data.map(collection => collection.name);
    
    console.log('现有集合:', collectionNames);
    
    // 需要确保存在的集合列表
    const requiredCollections = [
      'users', 
      'images', 
      'user_likes', 
      'user_favorites', 
      'comments'
    ];
    
    // 检查并创建不存在的集合
    for (const collName of requiredCollections) {
      if (!collectionNames.includes(collName)) {
        console.log(`创建集合: ${collName}`);
        await db.createCollection(collName);
      }
    }
    
    return {
      success: true,
      message: '数据库集合初始化完成'
    };
  } catch (error) {
    console.error('数据库集合初始化失败:', error);
    return {
      success: false,
      message: '数据库集合初始化失败',
      error
    };
  }
}

module.exports = {
  ensureCollectionsExist
}; 